../../../../models/demos/wormhole/stable_diffusion/tests/test_basic_transformer_block.py